#第一周作业2 自定义计算目录大小的函数

#自定义计算目录大小的函数
import os
def dirsize(dir):
	'''
	返回值为size_sum:目录大小,初始值为0
	'''
	if not os.path.isdir(dir):
		return 'No input file exists.'
	size_sum = 0
	dlist = os.listdir(dir)
	cwdpath = os.path.abspath(dir)
	for i in dlist:
		file = os.path.join(cwdpath, i)
		if os.path.isfile(file):
			size = os.path.getsize(file)
			size_sum += size
		elif os.path.isdir(file):
			size_sum += dirsize(file)
			print(size_sum)
		else:
			print('The file is not exists.')
	return size_sum

print(dirsize('/home/webs/python/size'))